GPT-based training